### 测试运行的python环境
import os
import sys
import torch


if __name__=='__main__':
    print("which python:{}".format(sys.executable))
    print()
    print("Test torch is running on cuda: {}".format(torch.cuda.is_available()))